import java.io.*;
import java.lang.reflect.Method;

/**
 * 自定义类加载器
 *
 * 隔离加载类
 * 在某些框架内进行中间件与应用的模块隔离，把类加载到不同的环境。
 * 比如：阿里内某容器框架通过自定义类加载器确保应用中依赖的jar包不会影响到中间件运行时使用的jar包。
 * 再比如：Tomcat这类Web应用服务器，内部自定义了好几种类加载器，用于隔离同一个Web应用服务器上的不同应用程序。
 *
 * 修改类加载的方式
 * 类的加载模型并非强制，除Bootstrap外，其他的加载并非一定要引入，或者根据实际情况在某个时间点进行按需进行动态加载。
 *
 * 扩展加载源
 * 比如从数据库、网络、甚至是电视机机顶盒进行加载。
 *
 * 防止源码泄露
 * Java代码容易被编译和篡改，可以进行编译加密。那么类加载也需要自定义，还原加密的字节码。
 *
 * 作者：Shaw_Young
 * 链接：https://www.jianshu.com/p/7ee1bfe63216
 * 来源：简书
 * 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
 *
 */
public class MyClassLoader extends ClassLoader{

    private String path; //类加载路径
    private String classLoadName; //类加载器名称

    public MyClassLoader() {
    }

    //让系统类加载器作为该类的父加载器
    public MyClassLoader(String path, String classLoadName){
        super();
        this.path = path;
        this.classLoadName = classLoadName;
    }

    //指定当前类的父类加载器
    public MyClassLoader(ClassLoader parent, String path, String classLoadName){
        super(parent);
        this.path = path;
        this.classLoadName = classLoadName;
    }


    /**
     * 记录自定义的类,通过自定义加载器去加载
     * @param name
     * @return
     * @throws ClassNotFoundException
     */
    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException{
//        String myPath = "file:///Users/allen/Desktop/" + name.replace(".","/") + ".class";
//        System.out.println(myPath);
//        byte[] cLassBytes = null;
//        Path path = null;
//        try {
//            path = Paths.get(new URI(myPath));
//            cLassBytes = Files.readAllBytes(path);
//        } catch (IOException | URISyntaxException e) {
//            e.printStackTrace();
//        }
//        Class clazz = defineClass(name, cLassBytes, 0, cLassBytes.length);
//        return clazz;
        byte [] data = readClassFileToByteArray(name);
        return this.defineClass(name,data,0,data.length);
    }

    /**
     * 从文件目录获取.class字节文件数组
     * @param name
     * @return
     */
    private byte [] readClassFileToByteArray(String name){
        InputStream is = null;
        byte [] data = null;
        name =  name.replaceAll("\\.","/");
        String filePath = this.path + path + ".class";
        File file =  new File(filePath);
        ByteArrayOutputStream os = new ByteArrayOutputStream();
        try{
            is = new FileInputStream(file);
            int temp = 0;
            while((temp = is.read()) != -1){
                os.write(temp);
            }
            data = os.toByteArray();
        }catch (Exception e){
            e.printStackTrace();
        }finally {
            try {
                if (is != null){
                    is.close();
                }
                if (os != null){
                    os.close();
                }
            }catch (IOException e){
                e.printStackTrace();
            }
        }
        return data;
    }

    public static void main(String[] args) throws ClassNotFoundException {
        MyClassLoader loader = new MyClassLoader();
        Class<?> aClass = loader.findClass("cn.lijie.MyTest");
        try {
            Object obj = aClass.newInstance();
            Method method = aClass.getMethod("show");
            method.invoke(obj);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

}
